<?php
    // +----------------------------------------------------------------------
    // | ThinkPHP [ WE CAN DO IT JUST THINK ]
    // +----------------------------------------------------------------------
    // | Copyright (c) 2006~2015 http://thinkphp.cn All rights reserved.
    // +----------------------------------------------------------------------
    // | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
    // +----------------------------------------------------------------------
    // | Author: yunwuxin <448901948@qq.com>
    // +----------------------------------------------------------------------
    namespace think\console\output;

    use think\console\output\formatter\Stack as StyleStack;
    use think\console\output\formatter\Style;

    class Formatter {
        private $decorated = false;
        private $styles    = [];
        private $styleStack;

        /**
         * 初始化命令行输出格式
         */
        public function __construct() {
            $this->setStyle('error', new Style('white', 'red'));
            $this->setStyle('info', new Style('green'));
            $this->setStyle('comment', new Style('yellow'));
            $this->setStyle('question', new Style('black', 'cyan'));
            $this->setStyle('highlight', new Style('red'));
            $this->setStyle('warning', new Style('black', 'yellow'));
            $this->styleStack = new StyleStack();
        }

        /**
         * 添加一个新样式
         * @param string $name  样式名
         * @param Style  $style 样式实例
         */
        public function setStyle($name, Style $style) {
            $this->styles[strtolower($name)] = $style;
        }

        /**
         * 转义
         * @param string $text
         * @return string
         */
        public static function escape($text) {
            return preg_replace('/([^\\\\]?)</is', '$1\\<', $text);
        }

        /**
         * 获取样式
         * @param string $name
         * @return Style
         * @throws \InvalidArgumentException
         */
        public function getStyle($name) {
            if (!$this->hasStyle($name)) {
                throw new \InvalidArgumentException(sprintf('Undefined style: %s', $name));
            }
            return $this->styles[strtolower($name)];
        }

        /**
         * 是否有这个样式
         * @param string $name
         * @return bool
         */
        public function hasStyle($name) {
            return isset($this->styles[strtolower($name)]);
        }

        /**
         * 使用所给的样式格式化文字
         * @param string $message 文字
         * @return string
         */
        public function format($message) {
            $offset   = 0;
            $output   = '';
            $tagRegex = '[a-z][a-z0-9_=;-]*';
            preg_match_all("#<(($tagRegex) | /($tagRegex)?)>#isx", $message, $matches, PREG_OFFSET_CAPTURE);
            foreach ($matches[0] as $i => $match) {
                $pos  = $match[1];
                $text = $match[0];
                if (0 != $pos && '\\' == $message[$pos - 1]) {
                    continue;
                }
                $output .= $this->applyCurrentStyle(substr($message, $offset, $pos - $offset));
                $offset = $pos + strlen($text);
                if ($open = '/' != $text[1]) {
                    $tag = $matches[1][$i][0];
                } else {
                    $tag = isset($matches[3][$i][0]) ? $matches[3][$i][0] : '';
                }
                if (!$open && !$tag) {
                    // </>
                    $this->styleStack->pop();
                } elseif (false === $style = $this->createStyleFromString(strtolower($tag))) {
                    $output .= $this->applyCurrentStyle($text);
                } elseif ($open) {
                    $this->styleStack->push($style);
                } else {
                    $this->styleStack->pop($style);
                }
            }
            $output .= $this->applyCurrentStyle(substr($message, $offset));
            return str_replace('\\<', '<', $output);
        }

        /**
         * 从堆栈应用样式到文字
         * @param string $text 文字
         * @return string
         */
        private function applyCurrentStyle($text) {
            return $this->isDecorated() && strlen($text) > 0 ? $this->styleStack->getCurrent()->apply($text) : $text;
        }

        /**
         * 获取外观标识
         * @return bool
         */
        public function isDecorated() {
            return $this->decorated;
        }

        /**
         * 设置外观标识
         * @param bool $decorated 是否美化文字
         */
        public function setDecorated($decorated) {
            $this->decorated = (bool)$decorated;
        }

        /**
         * 根据字符串创建新的样式实例
         * @param string $string
         * @return Style|bool
         */
        private function createStyleFromString($string) {
            if (isset($this->styles[$string])) {
                return $this->styles[$string];
            }
            if (!preg_match_all('/([^=]+)=([^;]+)(;|$)/', strtolower($string), $matches, PREG_SET_ORDER)) {
                return false;
            }
            $style = new Style();
            foreach ($matches as $match) {
                array_shift($match);
                if ('fg' == $match[0]) {
                    $style->setForeground($match[1]);
                } elseif ('bg' == $match[0]) {
                    $style->setBackground($match[1]);
                } else {
                    try {
                        $style->setOption($match[1]);
                    } catch (\InvalidArgumentException $e) {
                        return false;
                    }
                }
            }
            return $style;
        }

        /**
         * @return StyleStack
         */
        public function getStyleStack() {
            return $this->styleStack;
        }
    }
